import streamlit as st
import data.Data as dd
import time
import getpass
# import re
st.set_page_config(
    page_title="私人助手登录页面",
    page_icon="😊"
)
st.title('Login  to your personal assistant')
username = st.text_input('请输入你的账号')
password = st.text_input('请输入你的密码',type='password')
loginFlag = st.button('登录')
# passwd = getpass.getpass()

def loginer(username, password):
    if username and password:
            # 4.查询数据库是否有重复信息
            if dd.query_login_no_exits(username):
                    # 3.手机号长度8位且输入密码一致
                res = dd.query_password_exits(username)
                # st.write(res)
                if res ["password"] == password:
                        st.success("登录成功")
                        time.sleep(2)
                        st.session_state.user_id = res["user_id"]
                        st.session_state.username = res["username"]
                        st.switch_page("pages/chatbot.py")
                else:
                    st.error("用户存在，密码不正确,请重新输入")

            else:
                st.error("用户不存在，请前往注册")
                # time.sleep(2)
                # st.switch_page("pages/register.py")
    else:
        st.error("请务必填写相关登录信息")

    # st.error("用户不存在")
    # time.sleep(2)
    # st.switch_page("pages/register.py")



if loginFlag:
    loginer(username,password)
#注册按钮
registerFlag = st.button('没有账号，点击注册')
#点击之后跳转
if registerFlag:
    #streamlit有一个函数switch_page(py页面文件的路径)
    st.switch_page("pages/register.py")